package main

import (
	"GoAdmin/common"
	"github.com/gin-gonic/gin"
	_ "github.com/jinzhu/gorm/dialects/mysql" //这个一定要引入哦！！
	"github.com/spf13/viper"
	"log"
	"os"
)

func main() {
	InitConfig()

	db := common.InitDB()
	defer db.Close()

	r := gin.Default()
	r = CollectRoute(r)

	port := viper.GetString("server.port")
	if port != "" {
		panic(r.Run(":" + port))
	} else {
		panic(r.Run())
	}
}

func InitConfig() {
	//获取当前文件目录
	workDir, _ := os.Getwd()
	//设置要读取的文件名
	viper.SetConfigName("application")
	//设置要读取的文件类型
	viper.SetConfigType("yml")
	//设置文件的路径
	viper.AddConfigPath(workDir + "/config")
	err := viper.ReadInConfig()
	if err != nil {
		log.Printf("配置文件加载失败.... err : %v", err)
	}
}
